#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
创建测试Excel文件
"""
from openpyxl import Workbook
import os

def create_test_worktime_excel():
    """创建工时数据测试Excel文件"""
    wb = Workbook()
    ws = wb.active
    ws.title = "工时数据"
    
    # 设置表头 - 使用实际的字段名
    headers = ['统计周期', '所属项目', '系统', '需求类型', '状态', '总工时（人天）']
    for col, header in enumerate(headers, 1):
        ws.cell(row=1, column=col, value=header)
    
    # 添加测试数据
    test_data = [
        ['2024-01', '核心业务系统', '用户管理系统', '功能需求', '已完成', 5.5],
        ['2024-01', '核心业务系统', '订单管理系统', '优化需求', '进行中', 3.2],
        ['2024-01', '移动端项目', '支付系统', '功能需求', '已完成', 8.0],
        ['2024-02', '数据平台', '报表系统', '事务型需求', '已完成', 12.5],
        ['2024-02', '智能推荐', '算法引擎', '优化需求', '进行中', 15.2],
        ['2024-02', '智能推荐', '数据处理', '功能需求', '已完成', 6.8],
    ]
    
    for row_idx, row_data in enumerate(test_data, 2):
        for col_idx, value in enumerate(row_data, 1):
            ws.cell(row=row_idx, column=col_idx, value=value)
    
    # 保存文件
    filename = 'test_worktime_data.xlsx'
    wb.save(filename)
    print(f"已创建测试文件: {filename}")
    print(f"文件路径: {os.path.abspath(filename)}")
    print("\n文件内容:")
    print("表头:", headers)
    for i, row in enumerate(test_data):
        print(f"数据{i+1}:", row)

if __name__ == "__main__":
    create_test_worktime_excel()